#pragma once

#include <iostream>
#include <string>
#include <atomic>
#include <fstream>
#include <vector>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    const std::string temp_path = "./temp/"; // 临时文件所在的路径

    // TimeUtil 是与时间操作相关的类
    class TimeUtil
    {
    public:
        // 将时间戳转换成字符串
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }

        // 获得毫秒级时间戳
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            // 将秒和微妙转换成毫秒
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000); 
        }
    };

    // PathUntil 是用来构建文件路径的类
    class PathUtil
    {
    public:
        // 添加后缀名
        static std::string AddSuffix(const std::string& file_name, const std::string& suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }

        // 编译时生成的临时文件
        // 构建源文件路径+后缀的完整文件名
        // 1234 -> ./temp/1234.cpp
        static std::string Src(const std::string& file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }

        // 构建可执行程序路径+后缀的完整文件名
        // 1234 -> ./temp/1234.exe
        static std::string Exe(const std::string& file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        
        // 存储编译错误信息的文件
        static std::string CompilerError(const std::string& file_name)
        {
            return AddSuffix(file_name, ".compiler_error");
        }

        // 运行时生成的临时文件
        static std::string Stdin(const std::string& file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }

        static std::string Stdout(const std::string& file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }

        // 构建标准错误文件路径+后缀的完整文件名
        // 1234 -> ./temp/1234.stderr
        static std::string Stderr(const std::string& file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };

    // FileUtil 是用于文件操作的类
    class FileUtil
    {
    public:
        // 判断文件是否存在
        static bool IsFileExists(const std::string& file_name)
        {
            struct stat st;
            if(stat(file_name.c_str(), &st) == 0)
            {
                // 获取文件属性成功, 则说明文件存在
                return true;
            }
            return false; // 获取文件属性失败, 则说明文件不存在
        }

        // 形成唯一文件名(毫秒级时间戳 + 原子性递增值)
        static std::string UniqueFileName()
        {
            static std::atomic_uint id(0);
            id++;
            std::string ms = TimeUtil::GetTimeMs();
            std::string unique_id = std::to_string(id);
            return ms + "_" + unique_id;
        }

        // 形成临时源文件
        static bool WriteFile(const std::string& target, const std::string& code)
        {
            std::ofstream out(target);
            if(!out.is_open()) return false; // 打开文件失败

            out.write(code.c_str(), code.size()); // 写入用户提交的代码形成源文件
            out.close();
            return true; 
        }

        // 读取临时文件的内容
        static bool ReadFile(const std::string& target, std::string* content, bool keep = false)
        {
            std::ifstream in(target);
            if(!in.is_open()) return false; // 打开文件失败

            // 按行读取文件内容
            std::string line;
            // 细节一: getline 的返回值为 istream, 而 
            // istream 类重载 operator bool 函数, 能
            // 将返回值转换成 bool 值进行判断
            // 细节二: getline 函数不会保存 '\n' 的, 
            // 但有些时候需要保留 '\n', 那么可以通过 
            // keep 参数来决定是否保留 '\n', 其中 
            // false 表示不保留
            while(std::getline(in, line))
            {
                *content += line;
                if(keep) *content += "\n";
            }

            in.close(); // 关闭文件
            return true;
        } 
    };

    // 字符串操作类
    class StringUtil
    {
    public:
        /****************************
         * str: 输入型参数, 要切分的字符串
         * target: 输出型参数, 保存切分后的字符串
         * sep: 分割符
        *****************************/
        static void SplitString(const std::string& str, std::vector<std::string>* target, std::string sep)
        {
            // boost 字符串切分
            boost::split(*target, str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}